#!/usr/bin/env python3
"""
日期时间 MCP 服务器测试脚本
"""

import sys
import os
sys.path.insert(0, os.path.dirname(__file__))

from main import (
    get_current_datetime,
    get_current_date,
    get_current_time,
    get_timestamp,
    get_iso_datetime,
    get_weekday,
    format_datetime
)

def test_all_functions():
    """测试所有日期时间函数"""
    print("=== 日期时间 MCP 服务器功能测试 ===\n")

    print("1. 当前日期时间:")
    print(f"   {get_current_datetime()}\n")

    print("2. 当前日期:")
    print(f"   {get_current_date()}\n")

    print("3. 当前时间:")
    print(f"   {get_current_time()}\n")

    print("4. 时间戳:")
    print(f"   {get_timestamp()}\n")

    print("5. ISO 格式:")
    print(f"   {get_iso_datetime()}\n")

    print("6. 星期几:")
    print(f"   {get_weekday()}\n")

    print("7. 自定义格式:")
    print(f"   美式格式: {format_datetime('%m/%d/%Y %I:%M:%S %p')}")
    print(f"   欧式格式: {format_datetime('%d.%m.%Y %H:%M')}")
    print(f"   简单格式: {format_datetime('%Y/%m/%d')}\n")

    print("=== 测试完成 ===")

if __name__ == "__main__":
    test_all_functions()
